from openai import OpenAI

# 创建 OpenAI 客户端，指向 Ollama 本地服务
client = OpenAI(
    base_url="http://localhost:11434/v1",  # Ollama 本地服务地址
    api_key="ollama-local"  # 必填，但随便填（Ollama 不验证）
)

# 替换为你本地已有的模型名称
response = client.chat.completions.create(
    model="deepseek-r1:1.5b",
    messages=[
        {"role": "user", "content": "讲一个程序员的冷笑话"}
    ]
)
print(response.choices[0].message.content)
